#!/usr/bin/env python
# -*- coding:utf-8 -*-

# file:do_exam1.py
# author:王雯雯
# datetime:2024-11-30
# software: PyCharm

import time


def timeit(func):

    def wrapper(*args, **kwargs):
        # 记录函数开始运行的时间
        start_time = time.time()
        # 调用原始函数，并获取其返回值
        result = func(*args, **kwargs)
        # 记录函数结束运行的时间
        end_time = time.time()
        print(f"{func.__name__} 函数运行时间: {end_time - start_time} 秒")
        return result
    return wrapper
